# 创建管理员序列化器类
from rest_framework import serializers
from apps.users.models import User


class AdminAuthSerializer(serializers.ModelSerializer):
    """管理员登录序列化器类"""
    # 重定义username, 不需要自动生成
    username = serializers.CharField(label='用户名')
    token = serializers.CharField(label='JWT Token', read_only=True)

    class Meta:
        model = User
        fields = ('id', 'username', 'password', 'token')

        extra_kwargs = {
            'password': {
                'write_only': True
            }
        }

    # 获取参数并进行校验(validate校验多个参数)
    def validate(self, attrs):
        """
        用户名和密码是否正确
        :param attrs: 传递给data的数据
        :return:
        """
        # 获取username和password
        username = attrs['username']
        password = attrs['password']

        # 进行用户名和密码校验
        try:
            # 根据username查询管理员用户
            user = User.objects.get(username=username, is_staff=True)
        except User.DoesNotExist:
            # 管理员用户不存在
            raise serializers.ValidationError('用户名或密码错误')
        else:
            # 校验密码是否正确
            if not user.check_password(password):
                raise serializers.ValidationError('用户名或密码错误')

            # 给attrs中添加user属性,保存登录用户(将user对象添加到attrs字典中)
            attrs['user'] = user

        # 必须返回
        return attrs

    # 重写create方法,服务器生成jwt token
    def create(self, validated_data):
        """生成JWT Token"""
        # username = validated_data.get('username')
        # user = User.objects.get(username=username)
        # 代码优化:因为上面已经查到了用户对象,validated_data包含attrs
        user = validated_data.get('user')

        # 服务器生成jwt token, 保存当前用户的身份信息
        from rest_framework_jwt.settings import api_settings

        # 组织payload数据的方法
        jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
        # 生成jwt token数据的方法
        jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER

        # # 组织payload数据
        payload = jwt_payload_handler(user)
        # 生成jwt token
        token = jwt_encode_handler(payload)

        # 临时给user对象增加属性,保存jwt token的数据
        user.token = token

        return user


class UserSerializer(serializers.ModelSerializer):
    """用户序列化器类"""

    class Meta:
        model = User
        fields = ('id', 'username', 'mobile', 'email')
